# 给定一个整数数组 nums，处理以下类型的多个查询:
#  计算索引 left 和 right （包含 left 和 right）之间的 nums 元素的 和 ，其中 left <= right
#
#  实现 NumArray 类：
#  NumArray(int[] nums) 使用数组 nums 初始化对象
#  int sumRange(int i, int j) 返回数组 nums 中索引 left 和 right 之间的元素的 总和 ，
#  包含 left 和right 两点（也就是 nums[left] + nums[left + 1] + ... + nums[right] )
#
#  示例 1：
# 输入：
# ["NumArray", "sumRange", "sumRange", "sumRange"]
# [[[-2, 0, 3, -5, 2, -1]], [0, 2], [2, 5], [0, 5]]
# 输出：
# [null, 1, -1, -3]
#
# 解释：
# NumArray numArray = new NumArray([-2, 0, 3, -5, 2, -1]);
# numArray.sumRange(0, 2); // return 1 ((-2) + 0 + 3)
# numArray.sumRange(2, 5); // return -1 (3 + (-5) + 2 + (-1))
# numArray.sumRange(0, 5); // return -3 ((-2) + 0 + 3 + (-5) + 2 + (-1))
from typing import List


class NumArray:
    """
    前缀和 prefixSum[i] 表示前 i 项和(i从1开始  prefixSum[0] = 0)
    则：prefixSum[j] - prefixSum[i] = nums[i + 1] + nums[i + 2] + ... + nums[j]
    """
    def __init__(self, nums: List[int]):
        self.prefixSum = [0] + [0] * len(nums)
        for i in range(1, len(nums) + 1):
            self.prefixSum[i] = self.prefixSum[i - 1] + nums[i - 1]

    def sumRange(self, left: int, right: int) -> int:
        return self.prefixSum[right + 1] - self.prefixSum[left]


if __name__ == '__main__':
    nums = [-2, 0, 3, -5, 2, -1]
    numArray = NumArray(nums)
    print(numArray.sumRange(0, 2))  # return 1 ((-2) + 0 + 3)
    print(numArray.sumRange(2, 5))  # return -1 (3 + (-5) + 2 + (-1))
    print(numArray.sumRange(0, 5))  # return -3 ((-2) + 0 + 3 + (-5) + 2 + (-1))
